tensor([[[[[0.0300, 0.0200],
           [0.0200, 0.0300]],

          [[0.1600, 0.0600],
           [0.0600, 0.1600]]],


         [[[0.1600, 0.0600],
           [0.0600, 0.1600]],

          [[0.9900, 0.9800],
           [0.0000, 0.0200]]]],



        [[[[0.4500, 0.4600],
           [0.4600, 0.4500]],

          [[0.4000, 0.4300],
           [0.4300, 0.4000]]],


         [[[0.4000, 0.4300],
           [0.4300, 0.4000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0500, 0.0500],
           [0.0500, 0.0500]],

          [[0.0200, 0.0600],
           [0.0600, 0.0200]]],


         [[[0.0200, 0.0600],
           [0.0600, 0.0200]],

          [[0.0100, 0.0200],
           [1.0000, 0.9800]]]],



        [[[[0.4700, 0.4800],
           [0.4800, 0.4700]],

          [[0.4200, 0.4600],
           [0.4600, 0.4200]]],


         [[[0.4200, 0.4600],
           [0.4600, 0.4200]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)